package com.atxyj.srb.oss.service.impl;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.CannedAccessControlList;
import com.atxyj.srb.oss.service.FileService;
import com.atxyj.srb.oss.util.OssProperties;
import org.joda.time.DateTime;
import org.springframework.stereotype.Service;
import java.io.InputStream;
import java.util.UUID;

/**
 * @author: xyj
 * @date: 2021/8/1
 */
@Service
public class FileServiceImpl implements FileService {


    @Override
    public String uploadFile(InputStream inputStream, String module, String fileName) {
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(OssProperties.ENDPOINT, OssProperties.KEY_ID, OssProperties.KEY_SECRET);
        // 判断bucketName是否存在
        if( !ossClient.doesBucketExist(OssProperties.BUCKET_NAME)) {
            ossClient.createBucket(OssProperties.BUCKET_NAME);
            ossClient.setBucketAcl(OssProperties.BUCKET_NAME, CannedAccessControlList.PublicRead);
        }
        // 上传文件
        // 生成文件目录结构 "image/2021/08/01/uuid.fileName"
        // 构建日期路径
        String timeFolder = new DateTime().toString("yyyy/MM/dd");

        //由于本地的fileName可能出现乱码什么的，所以根据原始文件名生成一个新的文件名，并且把原始文件的扩展名拿到
        fileName = UUID.randomUUID().toString() + fileName.substring(fileName.lastIndexOf("."));
        String key = module  + timeFolder + fileName;
        ossClient.putObject(OssProperties.BUCKET_NAME, key, inputStream);

        // 关闭OSSClient。
        ossClient.shutdown();

        //返回文件的url地址

        return "https://" + OssProperties.BUCKET_NAME + "." +OssProperties.ENDPOINT + "/" + key;
    }

    @Override
    public void removeFile(String url) {
        String host = "https://" + OssProperties.BUCKET_NAME + "." + OssProperties.ENDPOINT + "/";

        String objectName = url.substring(host.length());
        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(OssProperties.ENDPOINT, OssProperties.KEY_ID, OssProperties.KEY_SECRET);

        // 删除文件或目录。如果要删除目录，目录必须为空。
        ossClient.deleteObject(OssProperties.BUCKET_NAME, objectName);

        // 关闭OSSClient。
        ossClient.shutdown();
    }
}
